package com.wlm.Utils;

import java.io.*;
import java.util.zip.Deflater;
import java.util.zip.DeflaterOutputStream;  
import java.util.zip.Inflater;  
import java.util.zip.InflaterInputStream;

/**
 * 
 * <Description> ZLib压缩工具 
 *  
 * @author STY <br>
 * @version 1.0 <br>
 * @CreateDate 2020年4月26日 <br>
 * @since V1.0 <br>
 * @see com.dlink.dview8.coreserver.utils <br>
 */
public class ZlibUtils {
    
    /**
     * 
     * Description: 压缩直接数组 
     *  
     * @author STY
     * @param dataStr
     * @return <br>
     */
    public static String compress(String dataStr) {
        byte[] data = dataStr.getBytes();
        byte[] output = new byte[0];  

        Deflater compresser = new Deflater();  

        compresser.reset();  
        compresser.setInput(data);  
        compresser.finish();  
        ByteArrayOutputStream bos = new ByteArrayOutputStream(data.length);  
        try {  
            byte[] buf = new byte[1024];  
            while (!compresser.finished()) {  
                int i = compresser.deflate(buf);  
                bos.write(buf, 0, i);  
            }  
            output = bos.toByteArray();  
        } catch (Exception e) {  
            output = data;  
            e.printStackTrace();  
        } finally {  
            try {  
                bos.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
        compresser.end();
        String outputStr = null;
        try {
            outputStr = new String(output, "ISO-8859-1");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        return outputStr;  
    }  

    /**
     * 
     * Description: 压缩 字节数组到输出流
     *  
     * @author STY
     * @param data
     * @param os <br>
     */
    public static void compress(byte[] data, OutputStream os) {  
        DeflaterOutputStream dos = new DeflaterOutputStream(os);  

        try {  
            dos.write(data, 0, data.length);  

            dos.finish();  

            dos.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        }  
    }  

    /**
     * 
     * Description: 解压缩 字节数组
     *  
     * @author STY
     * @param inputStr
     * @return <br>
     */
    public static String decompress(String inputStr) {
        byte[] data = new byte[0];
        try {
            data = inputStr.getBytes("ISO-8859-1");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        byte[] output = new byte[0];  

        Inflater decompresser = new Inflater();  
        decompresser.reset();  
        decompresser.setInput(data);  

        ByteArrayOutputStream o = new ByteArrayOutputStream(data.length);  
        try {  
            byte[] buf = new byte[1024];  
            while (!decompresser.finished()) {  
                int i = decompresser.inflate(buf);  
                o.write(buf, 0, i);  
            }  
            output = o.toByteArray();  
        } catch (Exception e) {  
            output = data;  
            e.printStackTrace();  
        } finally {  
            try {  
                o.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  

        decompresser.end();
        String outputStr = new String(output);
        return outputStr;  
    }  

    /**
     * 
     * Description: 解压缩 输入流 到字节数组
     *  
     * @author STY
     * @param is
     * @return <br>
     */
    public static byte[] decompress(InputStream is) {  
        InflaterInputStream iis = new InflaterInputStream(is);  
        ByteArrayOutputStream o = new ByteArrayOutputStream(1024);  
        try {  
            int i = 1024;  
            byte[] buf = new byte[i];  

            while ((i = iis.read(buf, 0, i)) > 0) {  
                o.write(buf, 0, i);  
            }  

        } catch (IOException e) {  
            e.printStackTrace();  
        }  
        return o.toByteArray();  
    }
    
    /** 
     * 压缩 
     *  
     * @param data 
     *            待压缩数据 
     * @return byte[] 压缩后的数据 
     */  
    public static byte[] compress(byte[] data) {  
        byte[] output = new byte[0];  
  
        Deflater compresser = new Deflater();  
  
        compresser.reset();  
        compresser.setInput(data);  
        compresser.finish();  
        ByteArrayOutputStream bos = new ByteArrayOutputStream(data.length);  
        try {  
            byte[] buf = new byte[1024];  
            while (!compresser.finished()) {  
                int i = compresser.deflate(buf);  
                bos.write(buf, 0, i);  
            }  
            output = bos.toByteArray();  
        } catch (Exception e) {  
            output = data;  
            e.printStackTrace();  
        } finally {  
            try {  
                bos.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
        compresser.end();  
        return output;  
    } 
    
    /** 
     * 解压缩 
     *  
     * @param data 
     *            待压缩的数据 
     * @return byte[] 解压缩后的数据 
     */  
    public static byte[] decompress(byte[] data) {  
        byte[] output = new byte[0];  
  
        Inflater decompresser = new Inflater();  
        decompresser.reset();  
        decompresser.setInput(data);  
  
        ByteArrayOutputStream o = new ByteArrayOutputStream(data.length);  
        try {  
            byte[] buf = new byte[1024];  
            while (!decompresser.finished()) {  
                int i = decompresser.inflate(buf);  
                o.write(buf, 0, i);  
            }  
            output = o.toByteArray();  
        } catch (Exception e) {  
            output = data;  
            e.printStackTrace();  
        } finally {  
            try {  
                o.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
  
        decompresser.end();  
        return output;  
    }  
}  
